"""
mysqlclient ---> import MySQLdb
pymysql ---> import pymysql
"""
import MySQLdb

no = int(input('部门编号: '))
name = input('部门名称: ')
location = input('部门所在地: ')

# 第一步：创建连接对象
conn = MySQLdb.connect(host='47.104.31.138', port=3306,
                       database='hrs', charset='utf8',
                       user='luohao', password='Luohao.618')
try:
    # 第二步：获取游标对象
    with conn.cursor() as cursor:
        # 第三步：通过游标对象向数据库发出SQL语句
        # execute方法的第一个参数是发给数据库的SQL语句
        # 如果SQL语句中有占位符，可以通过第二个参数传入元组
        # 用元组中的值去替换掉占位符即可
        result = cursor.execute(
            'insert into tb_dept values (%s, %s, %s)',
            (no, name, location)
        )
        if result == 1:
            print('新增部门成功!!!')
    # 第四步：操作成功，提交之前的操作
    conn.commit()
except MySQLdb.MySQLError as err:
    print(err)
    # 第四步：操作失败，回滚之前的操作
    conn.rollback()
finally:
    # 第五步：关闭连接，释放资源
    conn.close()
